from flask_wtf import FlaskForm
from flask_wtf.file import FileAllowed, FileRequired, FileField, FileSize
from wtforms import SubmitField


class ImageUpload(FlaskForm):
    photo = FileField('Upload Image',
                      validators=[FileRequired(), FileSize(max_size=3 * 1024 * 1024)])
    # FileAllowed(['.jpg', '.jpeg', '.PNG', '.png', '.gif'])
    submit = SubmitField()
